import os
import subprocess

IN_DATA_DIR = "E:/AVA/videos"
OUT_DATA_DIR = "E:/AVA/videos_15min"

# 确保输出目录存在
if not os.path.exists(OUT_DATA_DIR):
    os.makedirs(OUT_DATA_DIR)

# 遍历输入目录中的所有视频文件
for video in os.listdir(IN_DATA_DIR):
    video_path = os.path.join(IN_DATA_DIR, video)
    out_name = os.path.join(OUT_DATA_DIR, video)

    # 确保输出文件不存在才进行转换
    if not os.path.isfile(out_name):
        # 使用ffmpeg提取从900秒开始的15分钟片段
        cmd = [
            "ffmpeg",
            "-ss", "900",
            "-t", "900",  # 注意：这里我们使用900秒而不是901秒，因为ffmpeg的-t参数指的是持续时间
            "-i", video_path,
            out_name
        ]
        subprocess.run(cmd)

print("Processing completed.")